Skip to content

Feature/jax point source#355

Merged
Jammy2211 merged 50 commits intomainfrom
feature/jax_point_source
Jun 16, 2025
Merged

Feature/jax point source#355
Jammy2211 merged 50 commits intomainfrom
feature/jax_point_source

Conversation

@Jammy2211
Copy link
Copy Markdown
Collaborator

Full support for JAX point source likelihood functions, including updates to OperateDeflections class.

@Jammy2211 Jammy2211 requested review from CKrawczyk and rhayes777 June 15, 2025 17:34
Copy link
Copy Markdown
Collaborator

@rhayes777 rhayes777 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice so it works end to end now?

@Jammy2211
Copy link
Copy Markdown
Collaborator Author

The grad function on the image-plane chi squareds gives all zeros, so I think there may still be a bug, but the jit compiles and gives a likelihood corrctly.

For a source plane chi squared the grad looks correct.

Need to dig deeper before deciding if its a bug.

@Jammy2211 Jammy2211 merged commit 661c4b9 into main Jun 16, 2025
0 of 8 checks passed
@Jammy2211 Jammy2211 deleted the feature/jax_point_source branch June 24, 2025 13:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants